import { NestFactory } from '@nestjs/core';
import rateLimit from 'express-rate-limit';
import express from 'express';
import helmet from 'helmet';
import Chalk from 'chalk';
import { join } from 'path';
import { mw } from 'request-ip';
import { NestExpressApplication } from '@nestjs/platform-express';
import { SwaggerModule, DocumentBuilder } from '@nestjs/swagger';
import { ValidationPipe, VersioningType } from '@nestjs/common';
import { ConfigService } from '@nestjs/config';
import session from 'express-session';
import { AppModule } from './app.module';
import { responseData } from './utils/result';
import {
  TransformOperatorInterceptor,
  TransformLogInterceptor,
} from './interceptor/transform.interceptor';
import { AllExceptionsFilter } from './filter/any-exception.filter';
import { Logger } from './utils/log4js';

async function bootstrap() {
  const app = await NestFactory.create<NestExpressApplication>(AppModule, {
    cors: true, // 启用 CORS
  });

  const config = app.get(ConfigService);

  app.useStaticAssets(join(__dirname, '..', 'www')); // 静态资源
  app.useStaticAssets(join(__dirname, '..', 'upload'), {
    prefix: '/upload',
  }); // 上传文件

  // 版本配置
  app.enableVersioning({
    type: VersioningType.URI,
    defaultVersion: '1',
  });

  app.use(
    session({
      secret: 'my-secret',
      resave: false,
      saveUninitialized: false,
    }),
  );

  // 设置访问频率
  app.use(
    rateLimit({
      windowMs: 10 * 60 * 1000,
      max: 1000, // 限制10分钟内最多只能访问1000次
      message: '请求过于频繁，请稍后再试',
      handler: (request, response, next, options) =>
        response
          .status(options.statusCode)
          .send(
            responseData({ code: 50001, language: request.get('language') }),
          ), // 重写返回数据格式
    }),
  );

  app.use(helmet()); // 通过适当地设置 HTTP 头，Helmet 可以帮助保护应用免受一些众所周知的 Web 漏洞的影响

  app.setGlobalPrefix(config.get('prefix')); // 设置前缀

  app.useGlobalPipes(
    new ValidationPipe({
      transform: true, // 基础类型会自动转换
      whitelist: true,
    }),
  ); // 自动验证

  // 获取真实 ip
  app.use(mw({ attributeName: 'ip' }));

  app.use(express.json()); // 解析表单中的 JSON 格式的数据
  app.use(express.urlencoded({ extended: true })); // 解析 URL-encoded 格式的请求体数据

  // 使用全局拦截器打印出参
  app.useGlobalInterceptors(new TransformLogInterceptor());
  // 过滤处理http异常
  app.useGlobalFilters(new AllExceptionsFilter());

  // Swagger文档配置, 必须放到setGlobalPrefix()方法之后
  const options = new DocumentBuilder()
    .setTitle('api接口文档')
    .setDescription('V1.0 接口文档描述')
    .setVersion('V1.0')
    .addBearerAuth()
    .build();
  const document = SwaggerModule.createDocument(app, options);
  SwaggerModule.setup('/doc', app, document);

  app.useGlobalInterceptors(new TransformOperatorInterceptor());

  await app.listen(3000);
  Logger.log(
    Chalk.green(`Nest 服务启动成功 `),
    `http://localhost:${config.get('port')}`,
    '\n',
    Chalk.green('swagger 文档地址        '),
    `http://localhost:${config.get('port')}/doc`,
  );
}
bootstrap();
